#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Author: '习惯'
# @Email: 'songbing513@gmail.com'
# @Date: '2025/8/23 22:53'
# @Project: 'McpAgent'
# @File: 'hello.py'
# @Software: 'PyCharm'


from langchain_openai import ChatOpenAI # 注意：这里仍然从 langchain_openai 导入
import os

# 2. 初始化指向 DeepSeek 的 ChatOpenAI
llm = ChatOpenAI(
    model="deepseek-chat",  # 指定 DeepSeek 模型，例如 'deepseek-chat' 或 'deepseek-coder'
    base_url=os.environ.get("OPENAI_API_BASE"),  # 关键：替换为 DeepSeek 的 API 端点
    api_key=os.environ.get("OPENAI_API_KEY"),  # 从环境变量读取 API Key
    temperature=0.7
)

# 3. 调用方式与 OpenAI 完全一致
response = llm.invoke("帮我起一个饭店的名字")
print(response.content)